package main

import (
	"log"

	"github.com/gin-gonic/gin"
	"github.com/joho/godotenv"

	"ai-doctor/handlers"
	"ai-doctor/middleware"
)

func main() {
	// 加载环境变量
	err := godotenv.Load()
	if err != nil {
		log.Printf("警告: 无法加载 .env 文件: %v", err)
	}

	// 创建 Gin 路由
	r := gin.Default()

	// 配置中间件
	r.Use(middleware.Cors())

	// 静态文件服务
	r.Static("/ai-doctor/static", "./frontend")
	r.StaticFile("/ai-doctor", "./frontend/index.html")

	// API 路由
	api := r.Group("/ai-doctor/api")
	{
		api.POST("/consult", handlers.HandleConsultation)
		api.POST("/stream-consult", handlers.HandleStreamConsultation) // 新增流式咨询路由
	}

	// 启动服务器
	log.Println("启动服务器，监听端口 8080...")
	if err := r.Run(":8080"); err != nil {
		log.Fatalf("启动服务器失败: %v", err)
	}
}
